package com.leetcode.array_matrix;

/**
 * @author Dennis Li
 * @date 2020/7/22 21:28
 */
public class MatrixReshape_566 {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        int[][] ans = new int[r][c];
        int high = nums.length, width = nums[0].length;
        if(r * c != high * width) return nums;
        int index = 0;
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                // 第几行的数 -- 用 index / 每一行的数
                // 第几行的第几个 -- 用取余来计算
                ans[i][j] = nums[index / width][index % width];
                index++;
            }
        }
        return ans;
    }
}
